package tree.AVLTree;

import lombok.Data;
import lombok.NoArgsConstructor;

@Data
@NoArgsConstructor
public class Node {
    private int value;
    private Node left;
    private Node right;

    public Node(int value) {
        this.value = value;
    }

    @Override
    public String toString() {
        return "Node{" +
                "value=" + value +
                '}';
    }

    /**
     * 编写一个中序遍历的方法
     */
    public void infixOrder() {
        if (this.left != null) {
            this.left.infixOrder();
        }
        System.out.println(this);
        if (this.right != null) {
            this.right.infixOrder();
        }
    }

    /**
     * 编写add方法 尽量不让值重复
     */
    public void add(Node node) {
        if (node == null) {
            return;
        }
        if (this.getValue() == node.getValue()) {
            System.out.println("输入的值重复了");
            return;
        }
        if (node.value < this.value) {
            if (this.left != null) {
                this.left.add(node);
            } else {
                this.left = node;
            }
        }
        if (node.value > this.value) {
            if (this.right != null) {
                this.right.add(node);
            } else {
                this.right = node;
            }
        }
    }

    /**
     * 根据java垃圾回收机制，一个对象没有引用的时候，就会被回收
     */
    public int height() {
        return Math.max(left == null ? 0 : left.height(), right == null ? 0 : right.height()) + 1;
    }

    /**
     * 编写获得根节点左子树高度的方法
     */
    public int getLeftHeight() {
        return left == null ? 0 : left.height();
    }

    /**
     * 编写获得根节点右子树高度的方法
     */
    public int getRightHeight() {
        return right == null ? 0 : right.height();
    }

}
